package com.kotodama.infrastructure.mongodb;

import com.kotodama.infrastructure.tenant.context.TenantContext;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Optional;

/**
 * MongoDB仓储服务
 * 提供租户感知的MongoDB操作
 */
@Service
public class MongoRepositoryService {

    private final MongoTemplate mongoTemplate;

    public MongoRepositoryService(MongoTemplate mongoTemplate) {
        this.mongoTemplate = mongoTemplate;
    }

    /**
     * 保存文档
     *
     * @param entity 实体对象
     * @param <T>    实体类型
     * @return 保存后的实体
     */
    public <T> T save(T entity) {
        addTenantInfo(entity);
        return mongoTemplate.save(entity);
    }

    /**
     * 保存文档到指定集合
     *
     * @param entity         实体对象
     * @param collectionName 集合名称
     * @param <T>            实体类型
     * @return 保存后的实体
     */
    public <T> T save(T entity, String collectionName) {
        addTenantInfo(entity);
        return mongoTemplate.save(entity, collectionName);
    }

    /**
     * 根据ID查找文档
     *
     * @param id          文档ID
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 实体对象
     */
    public <T> Optional<T> findById(Object id, Class<T> entityClass) {
        Query query = createTenantAwareQuery(Query.query(Criteria.where("id").is(id)));
        T result = mongoTemplate.findOne(query, entityClass);
        return Optional.ofNullable(result);
    }

    /**
     * 根据ID查找文档（指定集合）
     *
     * @param id             文档ID
     * @param entityClass    实体类
     * @param collectionName 集合名称
     * @param <T>            实体类型
     * @return 实体对象
     */
    public <T> Optional<T> findById(Object id, Class<T> entityClass, String collectionName) {
        Query query = createTenantAwareQuery(Query.query(Criteria.where("id").is(id)));
        T result = mongoTemplate.findOne(query, entityClass, collectionName);
        return Optional.ofNullable(result);
    }

    /**
     * 查找所有文档
     *
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 实体列表
     */
    public <T> List<T> findAll(Class<T> entityClass) {
        Query query = createTenantAwareQuery(new Query());
        return mongoTemplate.find(query, entityClass);
    }

    /**
     * 根据条件查找文档
     *
     * @param query       查询条件
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 实体列表
     */
    public <T> List<T> find(Query query, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        return mongoTemplate.find(tenantAwareQuery, entityClass);
    }

    /**
     * 根据条件查找单个文档
     *
     * @param query       查询条件
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 实体对象
     */
    public <T> Optional<T> findOne(Query query, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        T result = mongoTemplate.findOne(tenantAwareQuery, entityClass);
        return Optional.ofNullable(result);
    }

    /**
     * 更新文档
     *
     * @param query       查询条件
     * @param update      更新操作
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 更新结果
     */
    public <T> boolean update(Query query, Update update, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        var result = mongoTemplate.updateFirst(tenantAwareQuery, update, entityClass);
        return result.getModifiedCount() > 0;
    }

    /**
     * 更新多个文档
     *
     * @param query       查询条件
     * @param update      更新操作
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 更新的文档数量
     */
    public <T> long updateMulti(Query query, Update update, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        var result = mongoTemplate.updateMulti(tenantAwareQuery, update, entityClass);
        return result.getModifiedCount();
    }

    /**
     * 删除文档
     *
     * @param query       查询条件
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 删除的文档数量
     */
    public <T> long delete(Query query, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        var result = mongoTemplate.remove(tenantAwareQuery, entityClass);
        return result.getDeletedCount();
    }

    /**
     * 根据ID删除文档
     *
     * @param id          文档ID
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 是否删除成功
     */
    public <T> boolean deleteById(Object id, Class<T> entityClass) {
        Query query = createTenantAwareQuery(Query.query(Criteria.where("id").is(id)));
        var result = mongoTemplate.remove(query, entityClass);
        return result.getDeletedCount() > 0;
    }

    /**
     * 统计文档数量
     *
     * @param query       查询条件
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 文档数量
     */
    public <T> long count(Query query, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        return mongoTemplate.count(tenantAwareQuery, entityClass);
    }

    /**
     * 检查文档是否存在
     *
     * @param query       查询条件
     * @param entityClass 实体类
     * @param <T>         实体类型
     * @return 是否存在
     */
    public <T> boolean exists(Query query, Class<T> entityClass) {
        Query tenantAwareQuery = createTenantAwareQuery(query);
        return mongoTemplate.exists(tenantAwareQuery, entityClass);
    }

    /**
     * 创建租户感知的查询
     *
     * @param originalQuery 原始查询
     * @return 租户感知的查询
     */
    private Query createTenantAwareQuery(Query originalQuery) {
        String tenantId = TenantContext.getTenantId();
        if (tenantId != null && !tenantId.trim().isEmpty()) {
            originalQuery.addCriteria(Criteria.where("tenantId").is(tenantId));
        }
        return originalQuery;
    }

    /**
     * 为实体添加租户信息
     *
     * @param entity 实体对象
     * @param <T>    实体类型
     */
    private <T> void addTenantInfo(T entity) {
        String tenantId = TenantContext.getTenantId();
        if (tenantId != null && !tenantId.trim().isEmpty()) {
            // 使用反射设置tenantId字段
            try {
                var field = entity.getClass().getDeclaredField("tenantId");
                field.setAccessible(true);
                field.set(entity, tenantId);
            } catch (NoSuchFieldException | IllegalAccessException e) {
                // 如果实体没有tenantId字段，忽略
            }
        }
    }
}
